how to write python database connection code